package month1;

import java.util.HashMap;
import java.util.Map;

// 560.和为 K 的子数组
// 给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。
//
// 子数组是数组中元素的连续非空序列。
// 示例 1：
//输入：nums = [1,1,1], k = 2
//输出：2
// 示例 2：
//输入：nums = [1,2,3], k = 3
//输出：2
public class SubarraySum560_03 {
    public static void main(String[] args) {
        System.out.println(subarraySum(new int[]{1,1,1}, 2));
    }

    /**
     * 前缀和hash计数
     * @param nums
     * @param k
     * @return
     */
    static int subarraySum(int[] nums, int k) {
        Map<Integer, Integer> count = new HashMap<>();
        count.put(0, 1);
        int sum = 0, res = 0;
        for(int num: nums) {
            sum += num;
            if (count.containsKey(sum - k)) {
                res += count.get(sum - k);
            }
            count.put(sum, count.getOrDefault(sum, 0) + 1);
        }
        return res;

    }
}
